package tech.lq0.dreamaticvoyage.entity;

import net.minecraft.sounds.SoundEvent;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.Mob;
import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
import net.minecraft.world.entity.ai.attributes.Attributes;
import net.minecraft.world.entity.animal.AbstractSchoolingFish;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.Level;
import tech.lq0.dreamaticvoyage.init.EntityRegistry;
import tech.lq0.dreamaticvoyage.init.ItemRegistry;

public class FukamizuFishEntity extends AbstractSchoolingFish {

    public FukamizuFishEntity(EntityType<? extends FukamizuFishEntity> type, Level level) {
        super(type, level);
    }

    public FukamizuFishEntity(Level level) {
        super(EntityRegistry.FUKAMIZU_FISH_ENTITY.get(), level);
    }

    @Override
    public int getMaxSchoolSize() {
        return 3;
    }

    @Override
    protected SoundEvent getFlopSound() {
        return SoundEvents.SALMON_FLOP;
    }

    @Override
    public ItemStack getBucketItemStack() {
        return new ItemStack(ItemRegistry.FUKAMIZU_FISH_BUCKET.get());
    }

    public static AttributeSupplier.Builder createAttributes() {
        return Mob.createMobAttributes().add(Attributes.MAX_HEALTH, 10.0D)
                .add(Attributes.ARMOR, 2.0D)
                .add(Attributes.ARMOR_TOUGHNESS, 1.0D);
    }
}
